MemFold: Learning Compact Soft Memory for Long-Context Personalization via On-Policy Optimization
MemFold:用在线策略优化学习紧凑软记忆的长上下文个性化
🧭 LLM 长期个性化记忆压缩与强化优化 ★ 深度解读
University of California, Santa Barbara · University of North Carolina at Chapel Hill · Harvard University · University of Wisconsin–Madison
固定预算软记忆+置信门控在线蒸馏,按下游行为而非文本重构优化长程个性化记忆
📌 工作参考价值(4 点)
- 用户长期画像/行为序列记忆:先用 writer 抽取 query 相关的结构化文本记忆,再压缩为固定数量 soft tokens(如 K=256),reader 只消费 K 个向量,推理成本不随历史长度线性增长;适合电商/广告 Agent 在长会话中保持个性化且控制 context 开销。
- 在线策略蒸馏:用冻结的 full-text reader 作为 teacher,只对学生自己采样 token 计算置信门控 σ(β(ℓ_T−ℓ_θ)) 来加权 student 的 token likelihood;无需 teacher 自回归生成,能对推荐文案/回复中局部违背用户偏好的 span 提供密集监督。
- 把标量 reward(点击/转化/偏好命中)与 token 级信号结合:GRPO 负责整体结果,OPD 定位“整体流畅但某个推荐项违背偏好”的错误;工程上同组 reward 方差接近 0 时把 advantages 置 0,避免无效更新。
- 训练期固定 memory snapshot 和 teacher,防止 RL 更新导致记忆漂移;上线前可用 shuffled/null memory 干预验证模型确实依赖实例特定记忆,而不是只背任务先验。